#!/bin/bash
module purge
module load  compilers/cuda/11.8 compilers/gcc/9.3.0 cudnn/8.4.0.27_cuda11.x anaconda
source activate torch2.1_cu118
export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128
export OMP_NUM_THREADS=1

python -u train-para.py config.yaml
### for test
# python -u test-para.py config.yaml ./ckpt/setoff-transolver-16slice-4layer/Dec10_13-16-26/best_model_transover_max_mae_0.0204752386.pt

